from utils.registry import Registry

# 模型注册器
MODELS = Registry('model')

# 导入具体的模型实现以确保注册
from .unet import UNetModel